Animations in python